import os
import shutil

# 设置源文件夹路径(使用原始字符串)
source_folder = r"E:\zhihao\2025文档\25年10月\ai_训练\ai项目_备份\yangzhihao\ai_project\public_security\traffic_accient\train\traffic_accidents_model\datasets\train\images"

# 创建两个新文件夹
lyh_folder = os.path.join(source_folder, "lyh")
hx_folder = os.path.join(source_folder, "hx")

# 创建文件夹
os.makedirs(lyh_folder, exist_ok=True)
os.makedirs(hx_folder, exist_ok=True)
print(f"已创建文件夹: {lyh_folder}")
print(f"已创建文件夹: {hx_folder}")

# 获取所有文件(排除包含 "yx" 的文件)
files_without_yx = []
for filename in os.listdir(source_folder):
    full_path = os.path.join(source_folder, filename)
    # 只处理文件,不处理文件夹
    if os.path.isfile(full_path):
        # 排除文件名中包含 "yx" 的文件
        if "yx" not in filename.lower():
            files_without_yx.append(filename)

print(f"\n找到 {len(files_without_yx)} 个不包含 'yx' 的文件")

# 一人一半分配
mid_point = len(files_without_yx) // 2

lyh_files = files_without_yx[:mid_point]
hx_files = files_without_yx[mid_point:]

print(f"分配给 lyh: {len(lyh_files)} 个文件")
print(f"分配给 hx: {len(hx_files)} 个文件")

# 复制文件到 lyh 文件夹
print("\n正在复制文件到 lyh 文件夹...")
for i, filename in enumerate(lyh_files, 1):
    src = os.path.join(source_folder, filename)
    dst = os.path.join(lyh_folder, filename)
    shutil.copy2(src, dst)
    if i % 10 == 0 or i == len(lyh_files):
        print(f"  已复制 {i}/{len(lyh_files)} 个文件")

# 复制文件到 hx 文件夹
print("\n正在复制文件到 hx 文件夹...")
for i, filename in enumerate(hx_files, 1):
    src = os.path.join(source_folder, filename)
    dst = os.path.join(hx_folder, filename)
    shutil.copy2(src, dst)
    if i % 10 == 0 or i == len(hx_files):
        print(f"  已复制 {i}/{len(hx_files)} 个文件")

print("\n✅ 完成!")
print(f"总结:")
print(f"  - lyh 文件夹: {len(lyh_files)} 个文件")
print(f"  - hx 文件夹: {len(hx_files)} 个文件")
print(f"  - 包含 'yx' 的文件保持不动")